import sys
import pickle
from numpy import random

# Ordem de argumentos:
# "Fwd Seg Size Avg", "Flow IAT Min", "Flow Duration", "Tot Fwd Pkts", "Pkt Size Avg", "Src Port", "Init Bwd Win Byts"
    # Fwd Seg Size Avg - Tamanho médio observado do fluxo de pacotes.
    # Flow IAT Min - Tempo mínimo entre dois fluxos de pacotes.
    # Flow Duration - Duração de um fluxo de pacotes.
    # Tot Fwd Pkts - Total de pacotes.
    # Pkt Size Avg - Tamanho médio por pacote.
    # Src Port - Porta de saída no hospedeiro de origem.
    # Init Bwd Win Byts - Bytes retornados ao remetente.
# argumentos = sys.argv[1:]
argumentos = random.randint(9999, size=(7))
descs = ["Fwd Seg Size Avg", "Flow IAT Min", "Flow Duration", "Tot Fwd Pkts", "Pkt Size Avg", "Src Port", "Init Bwd Win Byts"]

for i, val in enumerate(argumentos):
    print(str(descs[i]) + ': ' + str(val))

print()
print(pickle.load(open('./model_ml_fit.sav', 'rb')).predict([argumentos]))
